#!/usr/bin/env python3
# -*- coding: utf-8 -*-
##############################################
# @Author: DengLibin 榆霖
# @Date: Create in 2022-03-15 15:36:44
# @Description: Pandas 可以很方便的处理 CSV 文件
##############################################

import pandas as pd


def run():
    df = pd.read_csv('nba.csv')
    # to_string() 用于返回 DataFrame 类型的数据，如果不使用该函数，则输出结果为数据的前面 5 行和末尾 5 行，中间部分以 ... 代替
    print(df.to_string())
    print('-------------------------------------------------------------------------------------------------')
    # 读取前面 5 行
    # head( n ) 方法用于读取前面的 n 行，如果不填参数 n ，默认返回 5 行。
    print(df.head(5))
    
    print('-------------------------------------------------------------------------------------------------')
    # tail( n ) 方法用于读取尾部的 n 行，如果不填参数 n ，默认返回 5 行，空行各个字段的值返回 NaN。
    print(df.tail())
    
    
    
    print('-------------------------------------------------------------------------------------------------')
    # 也可以使用 to_csv() 方法将 DataFrame 存储为 csv 文件：
    # 三个字段 name, site, age
    nme = ["Google", "Runoob", "Taobao", "Wiki"]
    st = ["www.google.com", "www.runoob.com", "www.taobao.com", "www.wikipedia.org"]
    ag = [90, 40, 80, 98]
    
    # 字典
    dic = {'name': nme, 'site': st, 'age': ag}
        
    df = pd.DataFrame(dic)
    
    # 保存 dataframe
    df.to_csv('site.csv')
    print('ok...................')

if __name__ == '__main__':
   run()
